###################
###################
## create results tables
###################
###################

## clear environment
rm(list = ls())

library(tidyverse)
library(cshapes)
library(sf)
library(caret)
library(ranger)
library(iml)
library(metrica)
library(pdp)
## for parallel computing
library(parallel)
library(doParallel)
## for time keeping
library(tictoc)

tic()

cl <- makeCluster(parallelly::availableCores() - 4, setup_timeout = 0.5) ## convention to leave 1 core for OS
registerDoParallel()

# ## gadm for mapping admin region metrics
# ## get isos for mapping with gadm
# data <- readRDS("data/fulldata_adm1_africa.rds")
# isos <- c(unique(data$isocode1), "ESH")
# gadm.afr.adm1 <- NULL
# 
# for (i in isos) {
#   temp <- eval(st_read(paste0("data/gadm/admin1"), paste("gadm41", i, "1", sep = "_")))
#   gadm.afr.adm1 <- rbind(gadm.afr.adm1, temp)
# }
# rm(i, temp, data, isos)
# length(unique(gadm.afr.adm1$GID_0))
# ## this is necessary for plotting later
# # sf_use_s2(FALSE)


## create two objects to store all prediction metrics in for regression and classification
## for paper summary plots
# metrics.cla.all <- NULL
metrics.reg.all <- NULL

## create vectors for loops and calculations
col1 <- rep(c(rep("Gtw", 2), rep("Cov", 2), rep("Bm", 2), rep("Bm_Cov", 2), rep("Bm_Gtw", 2), rep("Cov_Gtw", 2), rep("Bm_Cov_Gtw", 2)), 2)
col2 <- rep(c(rep(c("Train", "Test"), 7)), 2)
col4 <- c("Gtw", "Cov", "Bm", "Bm_Gtw", "Cov_Gtw", "Bm_Cov_Gtw")

years <- 2020:2023
viol <- c("sbv", "osv", "nsv", "sri")
fw <- "fw3"

## for code testing
# i <- 2023
# j <- "sbv"
# k <- "fw3"
# l <- 1

for (i in years) {
  
  for (j in viol) {
    
    for (k in fw) {
      
      ## load all data for later
      for (l in 1:length(col1)) {
        print(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"))
        data.temp <- eval(readRDS(paste0("rds/predictions/", paste(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"), "rds", sep = "."))))
        assign(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"), data.temp)
      }
      rm(l)
      
      ## variable importance tables and plots
      ## hyperparameter tuning results plots
      ## pdp plots for gtw variables
      for (l in 1:length(col4)) {
        print(paste("rf_adm1_afr", i, j, tolower(col4[l]), k, sep = "_"))
        model.temp <- eval(readRDS(paste0("rds/ml_models/", paste(paste("rf_adm1_afr", i, j, tolower(col4[l]), k, sep = "_"), "rds", sep = "."))))

        ## pdp plots for gtw models
        if (col4[l] == "Gtw") {
          en_news <- model.temp %>%
            pdp::partial(pred.var = "hits_en_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          es_news <- model.temp %>%
            pdp::partial(pred.var = "hits_es_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          fr_news <- model.temp %>%
            pdp::partial(pred.var = "hits_fr_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          de_news <- model.temp %>%
            pdp::partial(pred.var = "hits_de_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          pt_news <- model.temp %>%
            pdp::partial(pred.var = "hits_pt_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          ru_news <- model.temp %>%
            pdp::partial(pred.var = "hits_ru_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          zh_news <- model.temp %>%
            pdp::partial(pred.var = "hits_zh_news") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          en_web <- model.temp %>%
            pdp::partial(pred.var = "hits_en_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          es_web <- model.temp %>%
            pdp::partial(pred.var = "hits_es_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          fr_web <- model.temp %>%
            pdp::partial(pred.var = "hits_fr_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          de_web <- model.temp %>%
            pdp::partial(pred.var = "hits_de_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          pt_web <- model.temp %>%
            pdp::partial(pred.var = "hits_pt_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          ru_web <- model.temp %>%
            pdp::partial(pred.var = "hits_ru_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          zh_web <- model.temp %>%
            pdp::partial(pred.var = "hits_zh_web") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          en_views <- model.temp %>%
            pdp::partial(pred.var = "views_en") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          es_views <- model.temp %>%
            pdp::partial(pred.var = "views_es") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          fr_views <- model.temp %>%
            pdp::partial(pred.var = "views_fr") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          de_views <- model.temp %>%
            pdp::partial(pred.var = "views_de") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          pt_views <- model.temp %>%
            pdp::partial(pred.var = "views_pt") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          ru_views <- model.temp %>%
            pdp::partial(pred.var = "views_ru") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)
          zh_views <- model.temp %>%
            pdp::partial(pred.var = "views_zh") %>%
            pdp::plotPartial(smooth = TRUE, lwd = 2)

          pdf(paste0("figures/pdp_plots/adm1_afr/", paste("pdp.adm1.afr", i, j, ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", ".")), tolower(col4[l])), k, "pdf", sep = ".")), width = 10, height = 6)
          gridExtra::grid.arrange(en_web, es_web, fr_web, de_web,
                                  pt_web, ru_web, zh_web,
                                  en_news, es_news, fr_news, de_news,
                                  pt_news, ru_news, zh_news,
                                  en_views, es_views, fr_views, de_views,
                                  pt_views, ru_views, zh_views,
                                  nrow = 3)
          dev.off()

        }
        rm(en_web, es_web, fr_web, de_web,
           pt_web, ru_web, zh_web,
           en_news, es_news, fr_news, de_news,
           pt_news, ru_news, zh_news,
           en_views, es_views, fr_views, de_views,
           pt_views, ru_views, zh_views)



        ## parameter tuning results plots
        plottitle <- paste0("Tuning results: ", ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", "+")), tolower(col4[l])), " (", i, " held out, ", ifelse(j == "sbv", "state-based violence fatalities, ", ifelse(j == "osv", "one-sided violence fatalities, ", ifelse(j == "nsv", "non-state violence fatalities, ", "security-related incidents, "))), ifelse(k == "fw5", "five year window", "three year window"), ")")
        toplot <- ggplot(model.temp) +
          ggtitle(plottitle) +
          theme_bw()

        ggsave(
          filename = paste0("figures/tuning_plots/adm1_afr/", paste("tuning.adm1.afr", i, j, ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", ".")), tolower(col4[l])), k, "pdf", sep = ".")),
          plot = toplot,
          width = 12,
          height = 6,
          dpi = 300
        )


        ## varimp plots
        plottitle <- paste0("Feature importance: ", ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", "+")), tolower(col4[l])), " (", i, " held out, ", ifelse(j == "sbv", "state-based violence fatalities, ", ifelse(j == "osv", "one-sided violence fatalities, ", ifelse(j == "nsv", "non-state violence fatalities, ", "security-related incidents, "))), ifelse(k == "fw5", "five year window", "three year window"), ")")
        toplot <- ggplot(varImp(model.temp)) +
          ggtitle(plottitle) +
          theme_bw()

        ggsave(
          filename = paste0("figures/varimp_plots/adm1_afr/", paste("varimp.adm1.afr", i, j, ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", ".")), tolower(col4[l])), k, "pdf", sep = ".")),
          plot = toplot,
          width = 12,
          height = 6,
          dpi = 300
        )


        ## varimp tables
        vimp <- as.data.frame(as.matrix(varImp(model.temp)$importance))%>%
          rownames_to_column("Feature")

        print(xtable::xtable(vimp %>%
                               arrange(desc(Overall)),
                             caption = paste0("Variable importance: ", ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", "+")), tolower(col4[l]))," (", i, " held-out)"),
                             label = paste("tab:varimp.adm1.afr", i, j, ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", ".")), tolower(col4[l])), k, sep = "."),
                             digits = 3),
              include.rownames = F,
              # hline.after = c(0, 0, 2, 4, 6, 8, 10, 12, 14),
              # floating.environment = "sidewaystable",
              file = paste0("tables/adm1_afr/", paste("varimp.adm1.afr", i, j, ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", ".")), tolower(col4[l])), k, "tex", sep = "."))
        )

      }
      rm(l, m)
      gc()



      ## create mean absolute error maps for all held out partitions

      data.container <- NULL
      for (l in 1:length(col4)) {
        print(paste("test_adm1_afr", i, j, tolower(col4[l]), k, sep = "_"))

        outcome <- ifelse(j == "sbv", "sbv_fat_be", ifelse(j == "osv", "osv_fat_be", ifelse(j == "nsv", "nsv_fat_be", "sri_num")))
        # data.temp.tocheck <- eval(readRDS(paste0("rds/predictions/", paste(paste("test", "cla", tolower(col4[l]), i, j, k, sep = "_"), "rds", sep = "."))))
        data.temp <- eval(readRDS(paste0("rds/predictions/", paste(print(paste("test_adm1_afr", i, j, tolower(col4[l]), k, sep = "_")), "rds", sep = ".")))) %>%
          group_by(isocode2full) %>%
          summarise(RMSE = sqrt(sum((eval(as.name(outcome)) - pred)^2)/length(eval(as.name(outcome)))),
                    MAE = sum(abs(eval(as.name(outcome)) - pred)/length(eval(as.name(outcome))))) %>%
          ungroup() %>%
          mutate(model = ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", "+")), tolower(col4[l])))
        data.container <- rbind(data.container, data.temp)

      }

      group.sf <- left_join(data.container, gadm.afr.adm1, , by = c("isocode2full" = "GID_1")) %>%
        st_as_sf()
      toplot <- ggplot(group.sf) +
        geom_sf(aes(fill = MAE)) +
        scale_fill_viridis_c(option = "plasma",
                             direction = -1) +
        facet_wrap(~factor(model, c("bm", "cov", "gtw", "bm+gtw", "cov+gtw", "bm+cov+gtw")), nrow = 1) +
        labs(caption = paste0("Mean absolute error for held-out ", i, " data by province.")) +
        ggtitle(paste0("Model performances predicting ", ifelse(j == "sbv", "state-based violence fatalities", ifelse(j == "osv", "one-sided violence fatalities", ifelse(j == "nsv", "non-state violence fatalities", "security-related incidents")))), subtitle = paste0("(", length(unique(group.sf$isocode2full)), " provinces)")) +
        theme_bw() +
        theme(plot.background = element_blank(),
              panel.grid.minor = element_blank(),
              panel.grid.major = element_blank(),
              panel.border = element_blank(),
              axis.ticks = element_blank(),
              axis.text = element_blank())

      ggsave(
        filename = paste0("figures/maps/adm1_afr/", paste("map.adm1.afr", i, j, k, "pdf", sep = ".")),
        plot = toplot,
        width = 12,
        height = 6,
        dpi = 300
      )

      rm(l, data.container, data.temp, group.sf, toplot)
      gc()


      ## create line plots for all held out partitions

      # data.container <- NULL
      # for (l in 1:length(col4)) {
      #   print(paste("test_adm1_afr", i, j, tolower(col4[l]), k, sep = "_"))
      #
      #   outcome <- ifelse(j == "sbv", "sbv_fat_be", ifelse(j == "osv", "osv_fat_be", ifelse(j == "nsv", "nsv_fat_be", "sri_num")))
      #   data.temp <- eval(readRDS(paste0("rds/predictions/", paste(print(paste("test_adm1_afr", i, j, tolower(col4[l]), k, sep = "_")), "rds", sep = ".")))) %>%
      #     ## only create lineplots for cases where at least one month saw a "positive" outcome
      #     group_by(isocode2full, year) %>%
      #     mutate(outcome_positive = any(eval(as.name(outcome)) > 0)) %>%
      #     ungroup() %>%
      #     filter(outcome_positive) %>%
      #     select(-outcome_positive) %>%
      #     group_by(isocode2full, year, month) %>%
      #     summarise(perror = pred - eval(as.name(outcome)),
      #               actual = eval(as.name(outcome)),
      #               pred = pred,
      #               ## keep province name for plotting title
      #               gadmname = gadmname) %>%
      #     # group_by(cowcode) %>%
      #     # summarise(RMSE = sqrt(sum((eval(as.name(outcome)) - pred)^2)/length(eval(as.name(outcome)))),
      #     #           MAE = sum(abs(eval(as.name(outcome)) - pred)/length(eval(as.name(outcome))))) %>%
      #     ungroup() %>%
      #     mutate(model = ifelse(str_detect(col4[l], "_"), tolower(str_replace_all(col4[l], "_", "+")), tolower(col4[l])))
      #   data.container <- rbind(data.container, data.temp)
      #
      # }
      # data.container$yearmonth <- as.Date(paste(data.container$month, "01", data.container$year, sep = "_"), format = "%m_%d_%Y")
      #
      # for (m in unique(data.container$gadmname)) {
      #   out <- ifelse(j == "sbv", "state-based violence fatalities", ifelse(j == "osv", "one-sided violence fatalities", ifelse(j == "nsv", "non-state violence fatalities", "security-related incidents")))
      #   toplot <- ggplot(subset(data.container, gadmname == m), aes(x = yearmonth)) +
      #     ## this one in combination with geom_label
      #     # geom_line(aes(group = model, colour = model, linetype = model), linewidth = 1) +
      #     geom_line(aes(y = pred, colour = model), linewidth = 1) +
      #     geom_point(aes(y = actual), color = "black", size = 2) +
      #     gghighlight::gghighlight(model %in% c("bm", "gtw", "bm+gtw")) +
      #     # geom_label(aes(label = model),
      #     #            data = subset(mydata.reg.all, country == countries) %>% filter(yearmonth == max(yearmonth) & model %in% c("bm", "wpc", "bm+wpc", "bm+cov+wpc"))) +
      #     # geom_line(aes(group = model, linetype = model), linewidth = 1) +
      #     labs(x = "Time",
      #          y = paste0("Actual and predicted ", out),
      #          caption = paste0("The plot shows the actual number of ",
      #                           out,
      #                           " as dots and the predicted number of ",
      #                           out,
      #                           " for the models as lines."),
      #          colour = "",
      #          linetype = "") +
      #     scale_x_date(labels = scales::date_format("%m-%Y"),
      #                  date_breaks = "2 months"
      #     ) +
      #     scale_y_continuous(breaks = ~round(unique(pretty(c(subset(data.container, gadmname == m)$perror, subset(data.container, gadmname == m)$actual))))) +
      #     # ggtitle("Prediction error of different models", subtitle = str_replace_all(paste(m), "_", " ")) +
      #     # ggtitle("Prediction error of different models", subtitle = ifelse(str_detect(paste(m), "(country)"), str_replace_all(paste(m), "_\\(country\\)", ""), str_replace_all(paste(m), "_", " "))) +
      #     ggtitle("Prediction error of different models", subtitle = paste(m)) +
      #     theme_bw() +
      #     theme(plot.background = element_blank(),
      #           panel.grid.minor = element_blank(),
      #           panel.grid.major = element_blank(),
      #           panel.border = element_blank())
      #
      #   ggsave(
      #     # filename = paste0("figures/line_plots/", paste("lp", "reg", i, "antigov", tolower(ifelse(str_detect(str_replace_all(paste(m), "_", "."), "(country)"), tolower(str_replace_all(str_replace_all(paste(m), "_", "."), "(country)", "")), tolower(str_replace_all(paste(m), "_", "."))), k, "pdf", sep = "."))),
      #     filename = paste0("figures/line_plots/adm1_afr/", paste("lp.adm1.afr", i, j, tolower(paste(m)), k, "pdf", sep = ".")),
      #     plot = toplot,
      #     width = 12, # replace with 10 if necessary
      #     height = 6,
      #     dpi = 300
      #   )
      #
      # }
      # ## the above may have to be uncommented again, not sure
      # rm(l, m, data.container, data.temp, toplot)
      
      
      

      
      
      
      ## create regression metrics table
      
      metrics.temp.reg <- as.data.frame(matrix(nrow = 14, ncol = 12,
                                               dimnames = list(NULL, c("Model",
                                                                       "Data",
                                                                       "Observ's",
                                                                       "RMSE",
                                                                       "MAE",
                                                                       "R2",
                                                                       "PCC",
                                                                       "AC",
                                                                       "CCC",
                                                                       "RIA",
                                                                       "RAC",
                                                                       "MIC"))))
      
      
      for (l in 1:nrow(metrics.temp.reg)) {
        data.temp <- eval(as.name(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_")))
        print(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"))
        outcome <- ifelse(j == "sbv", "sbv_fat_be", ifelse(j == "osv", "osv_fat_be", ifelse(j == "nsv", "nsv_fat_be", "sri_num")))
        metrics.temp.reg[l,1] <- ifelse(str_detect(col1[l], "_"), tolower(str_replace_all(col1[l], "_", "+")), tolower(col1[l]))
        metrics.temp.reg[l,2] <- tolower(col2[l])
        metrics.temp.reg[l,3] <- nrow(data.temp)
        metrics.temp.reg[l,4] <- metrica::RMSE(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,5] <- metrica::MAE(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,6] <- metrica::R2(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,7] <- metrica::r(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,8] <- metrica::Xa(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,9] <- metrica::CCC(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,10] <- metrica::d1r(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,11] <- metrica::RAC(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg[l,12] <- metrica::MIC(data.temp, eval(as.name(outcome)), pred)
      }
      rm(l, outcome)
      
      
      
      print(xtable::xtable(metrics.temp.reg,
                           caption = paste0("Performance metrics (", i, " held-out): ", ifelse(j == "sbv", "state-based violence fatalities", ifelse(j == "osv", "one-sided violence fatalities", ifelse(j == "nsv", "non-state violence fatalities", "security-related incidents")))),
                           label = paste("tab:metrics.adm1.afr", i, j, k, sep = "."),
                           digits = 3),
            include.rownames = F,
            hline.after = c(0, 0, 2, 4, 6, 8, 10, 12, 14),
            # floating.environment = "sidewaystable",
            file = paste0("tables/adm1_afr/", paste("metrics.adm1.afr", i, j, k, "tex", sep = ".")))
      
      ## this is to replace [ht] with [!htbp] to make four tables fit on one page in the appendix
      input_dir <- "tables/adm1_afr/"
      file_name <- paste0("metrics.adm1.afr.", i, ".", j, ".", k, ".tex")
      file_path <- file.path(input_dir, file_name)
      
      # Read the file contents
      tex_lines <- readLines(file_path)
      
      # Replace the specific line containing "\begin{table}[ht]"
      tex_lines <- gsub("\\\\begin\\{table\\}\\[ht\\]", "\\\\begin{table}[!htbp]", tex_lines)
      
      # Write the modified contents back to the file
      writeLines(tex_lines, file_path)
      
      cat(paste("Updated:", file_name, "\n"))
      
      
      rm(metrics.temp.reg)
      
      
      
      # metrics.temp.reg.red <- as.data.frame(matrix(nrow = 14, ncol = 8,
      #                                              dimnames = list(NULL, c("Model",
      #                                                                      "Data",
      #                                                                      "Observ's",
      #                                                                      "RMSE",
      #                                                                      "MAE",
      #                                                                      # "R2",
      #                                                                      # "PCC",
      #                                                                      "AC",
      #                                                                      "CCC",
      #                                                                      # "RIA",
      #                                                                      "RAC"
      #                                                                      # "MIC"
      #                                              ))))
      # 
      # 
      # for (l in 1:nrow(metrics.temp.reg.red)) {
      #   data.temp <- eval(as.name(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_")))
      #   print(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"))
      #   outcome <- ifelse(j == "sbv", "sbv_fat_be", ifelse(j == "osv", "osv_fat_be", ifelse(j == "nsv", "nsv_fat_be", "sri_num")))
      #   metrics.temp.reg.red[l,1] <- ifelse(str_detect(col1[l], "_"), tolower(str_replace_all(col1[l], "_", "+")), tolower(col1[l]))
      #   metrics.temp.reg.red[l,2] <- tolower(col2[l])
      #   metrics.temp.reg.red[l,3] <- nrow(data.temp)
      #   metrics.temp.reg.red[l,4] <- metrica::RMSE(data.temp, eval(as.name(outcome)), pred)
      #   metrics.temp.reg.red[l,5] <- metrica::MAE(data.temp, eval(as.name(outcome)), pred)
      #   # metrics.temp.reg.red[l,6] <- metrica::R2(data.temp, eval(as.name(outcome)), pred)
      #   # metrics.temp.reg.red[l,7] <- metrica::r(data.temp, eval(as.name(outcome)), pred)
      #   metrics.temp.reg.red[l,6] <- metrica::Xa(data.temp, eval(as.name(outcome)), pred)
      #   metrics.temp.reg.red[l,7] <- metrica::CCC(data.temp, eval(as.name(outcome)), pred)
      #   # metrics.temp.reg.red[l,10] <- metrica::d1r(data.temp, eval(as.name(outcome)), pred)
      #   metrics.temp.reg.red[l,8] <- metrica::RAC(data.temp, eval(as.name(outcome)), pred)
      #   # metrics.temp.reg.red[l,12] <- metrica::MIC(data.temp, eval(as.name(outcome)), pred)
      # }
      # rm(l, outcome)
      # 
      # 
      # print(xtable::xtable(metrics.temp.reg.red,
      #                      caption = paste0("Performance metrics (", i, " held-out): ", ifelse(j == "sbv", "state-based violence fatalities", ifelse(j == "osv", "one-sided violence fatalities", ifelse(j == "nsv", "non-state violence fatalities", "security-related incidents")))),
      #                      label = paste("tab:metrics.red.adm1.afr", i, j, k, sep = "."),
      #                      digits = 3),
      #       include.rownames = F,
      #       hline.after = c(0, 0, 2, 4, 6, 8, 10, 12, 14),
      #       # floating.environment = "sidewaystable",
      #       file = paste0("tables/adm1_afr/", paste("metrics.red.adm1.afr", i, j, k, "tex", sep = ".")))
      # 
      # rm(metrics.temp.reg)
      
      
      ## plot for paper
      
      metrics.temp.reg.red <- as.data.frame(matrix(nrow = 14, ncol = 8,
                                                   dimnames = list(NULL, c("Model",
                                                                           "Data",
                                                                           "Observ's",
                                                                           "RMSE",
                                                                           "MAE",
                                                                           "AC",
                                                                           "CCC",
                                                                           "RAC"
                                                   ))))
      
      
      for (l in 1:nrow(metrics.temp.reg.red)) {
        data.temp <- eval(as.name(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_")))
        print(paste(tolower(col2[l]), "adm1_afr", i, j, tolower(col1[l]), k, sep = "_"))
        outcome <- ifelse(j == "sbv", "sbv_fat_be", ifelse(j == "osv", "osv_fat_be", ifelse(j == "nsv", "nsv_fat_be", "sri_num")))
        metrics.temp.reg.red[l,1] <- ifelse(str_detect(col1[l], "_"), tolower(str_replace_all(col1[l], "_", "+")), tolower(col1[l]))
        metrics.temp.reg.red[l,2] <- tolower(col2[l])
        metrics.temp.reg.red[l,3] <- nrow(data.temp)
        metrics.temp.reg.red[l,4] <- metrica::RMSE(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg.red[l,5] <- metrica::MAE(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg.red[l,6] <- metrica::Xa(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg.red[l,7] <- metrica::CCC(data.temp, eval(as.name(outcome)), pred)
        metrics.temp.reg.red[l,8] <- metrica::RAC(data.temp, eval(as.name(outcome)), pred)
      }
      rm(l, outcome)
      
      
      ## gather all prediction metrics for summary plot
      metrics.reg <- metrics.temp.reg.red %>%
        mutate(year = i,
               type = j,
               fw = k)
      
      if (is.null(metrics.reg.all)) {
        metrics.reg.all <- metrics.reg
      } else {
        metrics.reg.all <- rbind(metrics.reg.all, metrics.reg)
      }
      
      
      metricstoplot.reg.red <- metrics.temp.reg.red %>%
        select(-"Observ's") %>%
        pivot_longer(cols = !c(Data, Model),
                     names_to = "metric",
                     values_to = "value") %>%
        mutate(group = Model)
      
      metricsplot.reg.red <- ggplot(metricstoplot.reg.red, aes(x = value, y = factor(group, levels = c("bm+cov+gtw", "bm+gtw", "cov+gtw", "bm+cov", "gtw", "cov", "bm"))
      )) +
        geom_point() +
        facet_grid(factor(Data, levels = c("train", "test"))~factor(metric, levels = c("RMSE", "MAE", "AC", "CCC", "RAC")), scales = "free_x") +
        theme_bw() +
        labs(x = "Performance", y = NULL) +
        ggh4x::facetted_pos_scales(x = list(
          metricstoplot.reg.red$metric == "AC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0, 1, 0.2))),
          metricstoplot.reg.red$metric == "CCC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0, 1, 0.2))),
          metricstoplot.reg.red$metric == "RAC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0, 1, 0.2)))
        ))
      
      ggsave(
        filename = paste0("figures/metrics_plots/adm1_afr/", paste("mp.adm1.afr", i, j, k, "pdf", sep = ".")),
        metricsplot.reg.red, 
        width = 12,
        height = 6,
        dpi = 120
      )
      
      rm(metrics.temp.reg.red, metricstoplot.reg.red, metricsplot.reg.red, metrics.reg)
      
    }
    
  }
  
}
rm(i, j, k)



for (j in viol) {
  
  for (k in fw) {
    
    metrics.temp <- eval(as.name(paste("metrics", "reg", "all", sep = ".")))
    metricstoplot <- metrics.temp %>%
      filter(Data == "test" & fw == k & type == j) %>%
      select(-c("Observ's", "Data", "fw", "type")) %>%
      pivot_longer(cols = !c(year, Model),
                   names_to = "metric",
                   values_to = "value") %>%
      mutate(group = Model)
    
    metricsplot <- ggplot(metricstoplot) +
      geom_point(aes(x = value, y = factor(group, levels = c("bm+cov+gtw", "bm+gtw", "cov+gtw", "bm+cov", "gtw", "cov", "bm")))) +
      geom_point(data = filter(metricstoplot, group %in% c("bm", "gtw", "bm+gtw")), aes(x = value, y = factor(group)), shape = 18, colour = "orange", size = 3) +
      facet_grid(factor(year, levels = c("2023", "2022", "2021", "2020"))~factor(metric, levels = c("RMSE", "MAE", "AC", "CCC", "RAC")), scales = "free_x") +
      theme_bw() +
      labs(x = "Performance", y = NULL) +
      ggh4x::facetted_pos_scales(x = list(
        metricstoplot$metric == "RMSE" ~ scale_x_continuous(breaks = scales::pretty_breaks(n = 4)),
        metricstoplot$metric == "MAE" ~ scale_x_continuous(breaks = scales::pretty_breaks(n = 4)),
        metricstoplot$metric == "AC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0.2, 0.8, 0.2))),
        metricstoplot$metric == "CCC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0.2, 0.8, 0.2))),
        metricstoplot$metric == "RAC" ~ scale_x_continuous(limits = c(0, 1), breaks = c(seq(0.2, 0.8, 0.2)))
      ))
    

    ggsave(
      filename = paste0("figures/metrics_plots/adm1_afr/", paste("mp.adm1.afr", j, k, "pdf", sep = ".")),
      metricsplot, 
      width = 12,
      height = 6,
      dpi = 120
    )
    
  }
}
rm(j, k)





stopCluster(cl)
registerDoSEQ()

toc()
